from PIL import Image

def remove_white_background(input_image_path):
    # 读取WebP格式图片
    image = Image.open(input_image_path)

    # 将图像转换为RGBA模式（包含透明通道）
    image = image.convert('RGBA')

    # 获取图像的像素数据
    pixels = image.load()

    # 创建一个与原始图像尺寸相同的空白图像
    result = Image.new('RGBA', image.size)

    # 遍历图像的每个像素
    for i in range(image.width):
        for j in range(image.height):
            # 获取当前像素的RGBA值
            r, g, b, a = pixels[i, j]

            # 判断像素是否为白色（或接近白色）
            if r > 200 and g > 200 and b > 200:
                # 将近白色像素设置为透明
                result.putpixel((i, j), (r, g, b, 0))
            else:
                # 将非白色像素保留
                result.putpixel((i, j), (r, g, b, a))

    # 保存抠图结果，覆盖原始图片
    result.save(input_image_path)
remove_white_background('test.png')